#浏览器工具函数

from selenium import webdriver

def get_browser(browser_type):
    """
    根据浏览器类型返回对应的 WebDriver 实例
    :param browser_type: 浏览器类型
    :return: WebDriver 实例
    """
    if browser_type == "chrome":
        return webdriver.Chrome()
    elif browser_type == "firefox":
        return webdriver.Firefox()
    else:
        raise ValueError("不支持的浏览器类型")